Draw straight line Python